/**
 * 标注任务类型相关的方法
 */

import { DmpSubTaskType, DmpTaskType } from '../types'

export const getTaskTypeName = (taskType: DmpTaskType | DmpSubTaskType) => {
  switch (taskType) {
    case DmpTaskType.img:
      return '图片'
    case DmpTaskType.text:
      return '文本'
    case DmpTaskType.audio:
      return '音频'
    case DmpTaskType.video:
      return '视频'
    case DmpSubTaskType.imgClass:
      return '图片分类'
    case DmpSubTaskType.imgObjDetect:
      return '目标检测'
    case DmpSubTaskType.imgSegmentation:
      return '图像分割'
    case DmpSubTaskType.textClass:
      return '文本分类'
    case DmpSubTaskType.textNamedEntity:
      return '命名实体'
    case DmpSubTaskType.textTriple:
      return '文本三元组'
    case DmpSubTaskType.textSimilarity:
      return '文本相似度'
    case DmpSubTaskType.audioClass:
      return '音频分类'
    case DmpSubTaskType.audioContentRecognition:
      return '内容识别'
    case DmpSubTaskType.audioSegmentation:
      return '语音分割'
    case DmpSubTaskType.videoFrame:
      return '视频逐帧标注'
    case DmpSubTaskType.videoSegment:
      return '视频分段标注'

    default:
      return '未知类型'
  }
}

// /* 一级分类  */
// export enum DmpTaskType {
//   // 图片
//   img = 1,
//   // 文本
//   text = 2,
//   // 音频
//   audio = 3,
//   // 音频
//   video = 4
// }

// /* 二级分类   */
// export enum DmpSubTaskType {
//   // 图片分类
//   imgClass = 101,
//   // 目标检测
//   imgObjDetect = 102,
//   // 图像分割
//   imgSegmentation = 103,

//   // 文本分类
//   textClass = 201,
//   // 命名实体
//   textNamedEntity = 202,
//   // 文本三元组
//   textTriple = 203,
//   // 文本相似度
//   textSimilarity = 204,

//   // 音频分类
//   audioClass = 301,
//   // 内容识别
//   audioContentRecognition = 302,
//   // 语音分割
//   audioSegmentation = 303,

//   // 视频逐帧标注
//   videoFrame = 401,
//   // 视频分段标注
//   videoSegment = 402
// }
